cmake_minimum_required(VERSION 3.20)

project(cutlass_hub)

set(CMAKE_CXX_STANDARD 17)

SET(THIRD_PARTY_DIR ${CMAKE_SOURCE_DIR}/third)
SET(CUTLASS_DIR ${THIRD_PARTY_DIR}/cutlass)

include_directories(${CUTLASS_DIR}/include)
include_directories(${CUTLASS_DIR}/tools/util/include)

# add_definitions(-g)
add_definitions(-O3)

include_directories(${CMAKE_SOURCE_DIR}/include)
set(CUDA_HOME /usr/local/cuda)

find_package(CUDA REQUIRED)
include_directories(${CUDA_INCLUDE_DIRS})
link_directories(${CUDA_LIBRARY_DIRS})


add_subdirectory(${THIRD_PARTY_DIR}/googletest)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/src)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/ut)